import os
from pathlib import Path

class TestConfig:
    # 基础配置
    BASE_DIR = Path(__file__).parent.parent
    
    # 默认测试配置
    DEFAULT_CONFIG = {
        'MYSQL_USER': 'test_user',
        'MYSQL_PASSWORD': 'test_password',
        'MYSQL_HOST': 'localhost',
        'MYSQL_DATABASE': 'test_db',
        'DEEPSEEK_API_KEY': 'sk-82fb38bf23d84e3292772f803f12d926',
        'DEEPSEEK_API_BASE': 'https://api.deepseek.com/v1'
    }
    
    @classmethod
    def load_config(cls):
        """加载配置"""
        # 首先尝试从 config.json 加载
        config_file = cls.BASE_DIR / 'config.json'
        if config_file.exists():
            import json
            with open(config_file) as f:
                return json.load(f)
        
        # 然后尝试从 .env 文件加载
        env_file = cls.BASE_DIR / '.env'
        if env_file.exists():
            from dotenv import load_dotenv
            load_dotenv(env_file)
            return {
                key: os.getenv(key, default)
                for key, default in cls.DEFAULT_CONFIG.items()
            }
        
        # 最后使用默认配置
        return cls.DEFAULT_CONFIG.copy() 